python branching and looping